package cn.swust.designpatternlearningplatform.domain.dto;

import lombok.Data;

@Data
public class FileDto {
    //文件名
    private String fileName;
    //文件数据
    private byte[] fileData;
    //文件后缀
    private String suffix;
    //该文件的上级目录
    private String parentFolder;

    /**
     * 创建pdf文件
     *
     * @param fileName
     * @param fileData
     * @param parentFile
     * @return cn.swust.designpatternlearningplatform.domain.dto.FileDto
     * @author Tang
     * @date 2023/10/15 16:33
     **/
    public static FileDto buildPdf(String fileName, byte[] fileData, String parentFile) {
        FileDto fileDto = buildPdf(fileName, fileData);
        if (parentFile != null && !parentFile.isEmpty()) {
            fileDto.setParentFolder(parentFile);
        }
        return fileDto;
    }

    public static FileDto buildPdf(String fileName, byte[] fileData) {
        FileDto fileDto = new FileDto();
        fileDto.setFileData(fileData);
        fileDto.setFileName(fileName);
        fileDto.setSuffix(".pdf");
        return fileDto;
    }
}
